Unit test in python programiz